"""
Models initialization module.

This module initializes model registries and imports necessary components.
"""

from core.registry.base import Registry
from models.interfaces import BackboneBase
from models.base import BaseModel

# 创建backbone和model注册表
BACKBONE_REGISTRY = Registry("backbone", BackboneBase)
MODEL_REGISTRY = Registry("model", BaseModel)

# 导出注册装饰器
register_backbone = BACKBONE_REGISTRY.register
register_model = MODEL_REGISTRY.register

# 导入所有模型和backbone以触发注册
# ...

# 确保所有模型都已注册
__all__ = [
    "BACKBONE_REGISTRY",
    "MODEL_REGISTRY",
    "register_backbone",
    "register_model",
]
